package future;

import java.util.concurrent.*;

/**
 * @author fei
 * @date 2023/1/4 16:24
 * @description 自动调用回调方法
 */
public class CompletableFutureUserDemo {

    public static void main(String[] args) {
        // defaultPool();
        ExecutorService threadPool = Executors.newFixedThreadPool(3);

        try {
            CompletableFuture.supplyAsync(() -> {
                System.out.println(Thread.currentThread().getName());
                int result = ThreadLocalRandom.current().nextInt(10);
                try{ TimeUnit.SECONDS.sleep(1);} catch (InterruptedException e) { e.printStackTrace(); }
                System.out.println("------1秒钟后出结果：" + result);
                if (result > 5) {
                    // 如果大于5，就抛异常
                    int zero = 10 / 0;
                }
                return result;
            }, threadPool).whenComplete((v, e) -> {
                // e 是异常，如果没有异常就输出结果 v
                if (e == null) {
                    System.out.println("------计算完成" + v);
                }
            }).exceptionally(e -> {
                e.printStackTrace();
                System.out.println("异常情况：" + e.getCause() + "\t" + e.getMessage());
                return null;
            });
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            threadPool.shutdown();
        }

        System.out.println(Thread.currentThread().getName() + " 线程先去忙其它任务");
    }


    /**
     * 使用默认的线程池 ForkJoinPool
     * 如果主线程结束，那么这个线程也就结束
     */
    private static void defaultPool() {
        CompletableFuture.supplyAsync(() -> {
            System.out.println(Thread.currentThread().getName());
            int result = ThreadLocalRandom.current().nextInt(10);
            try{ TimeUnit.SECONDS.sleep(1);} catch (InterruptedException e) { e.printStackTrace(); }
            System.out.println("------1秒钟后出结果：" + result);
            return result;
        }).whenComplete((v, e) -> {
            // e 是异常，如果没有异常就输出结果 v
            if (e == null) {
                System.out.println("------计算完成" + v);
            }
        }).exceptionally(e -> {
            e.printStackTrace();
            System.out.println("异常情况：" + e.getCause() + "\t" + e.getMessage());
            return null;
        });
        System.out.println(Thread.currentThread().getName() + " 线程先去忙其它任务");

        // 主线程不要立刻结束，否则 completableFuture 默认使用的线程池会立刻关闭，暂停 3秒钟线程
        // try{ TimeUnit.SECONDS.sleep(3);} catch (InterruptedException e) { e.printStackTrace(); }
    }

}
